python heapq example